# d16, 256x256
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5 torchrun --nproc_per_node=6 --nnodes=1 --node_rank=0 --master_addr=127.0.0.1 --master_port=1235 train.py \
  --data_path=/ifs/data/PlatforMax/common/Dataset/ImageNet/ImageNet2012 \
  --condition_path=/ifs/root/ipa01/101/user_101002/Datasets/ImageNet2012/train_gray \
  --vae_ckpt=/ifs/root/ipa01/101/user_101002/Download_Files/CAR_Ckpt/vae_ch160v4096z32.pth \
  --pretrained_var_ckpt=/ifs/root/ipa01/101/user_101002/Download_Files/CAR_Ckpt/var_d16.pth \
  --tblr=0.0001 --depth=16 --bs=114 --ep=200 --fp16=1 --alng=1e-3 --wpe=0.1 --workers=72